"""
Shawn
11/29/2018
"""


from PIL import Image
import argparse


parser = argparse.ArgumentParser()

parser.add_argument("file")
parser.add_argument("-o","--output")
parser.add_argument("--width", type = int, default = 80)
parser.add_argument("--height", type = int, default = 80)

args = parser.parse_args()

IMG = args.file
WIDTH = args.width
HEIGHT = args.height
OUTPUT = args.output

ascii_char = ["M","@","H","Q","$","O","C","?","7",">","!",";","."," "]

def get_char(r,g,b,alpha = 256):
    if alpha == 0:
        return '`'
    length = len(ascii_char)
    gray = int(0.299*r + 0.587*g + 0.114*b)
    unit = (256.0+1)/length
    return "{:1}".format(ascii_char[int(gray/unit)])

if __name__ == "__main__":

    im = Image.open(IMG)
    im = im.resize((WIDTH,HEIGHT),Image.NEAREST)

    txt = ""

    for i in range(HEIGHT):
        for j in range(WIDTH):
            txt += get_char(*im.getpixel((j,i)))
        txt += "\n"

    print(txt)

    if OUTPUT:
        with open(OUTPUT, "w") as f:
            f.write(txt)
    else:
        with open("output.txt","w") as f:
            f.write(txt)
